// Copyright (c) 2012 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "base/memory/aligned_memory.h"

#include <memory>

#include "build/build_config.h"
#include "testing/gtest/include/gtest/gtest.h"

#define EXPECT_ALIGNED(ptr, align) \
    EXPECT_EQ(0u, reinterpret_cast<uintptr_t>(ptr) & (align - 1))

namespace {

using base::AlignedMemory;

TEST(AlignedMemoryTest, StaticAlignment)
{
    static AlignedMemory<8, 8> raw8;
    static AlignedMemory<8, 16> raw16;
    static AlignedMemory<8, 256> raw256;
    static AlignedMemory<8, 4096> raw4096;

    EXPECT_EQ(8u, ALIGNOF(raw8));
    EXPECT_EQ(16u, ALIGNOF(raw16));
    EXPECT_EQ(256u, ALIGNOF(raw256));
    EXPECT_EQ(4096u, ALIGNOF(raw4096));

    EXPECT_ALIGNED(raw8.void_data(), 8);
    EXPECT_ALIGNED(raw16.void_data(), 16);
    EXPECT_ALIGNED(raw256.void_data(), 256);
    EXPECT_ALIGNED(raw4096.void_data(), 4096);
}

TEST(AlignedMemoryTest, StackAlignment)
{
    AlignedMemory<8, 8> raw8;
    AlignedMemory<8, 16> raw16;
    AlignedMemory<8, 128> raw128;

    EXPECT_EQ(8u, ALIGNOF(raw8));
    EXPECT_EQ(16u, ALIGNOF(raw16));
    EXPECT_EQ(128u, ALIGNOF(raw128));

    EXPECT_ALIGNED(raw8.void_data(), 8);
    EXPECT_ALIGNED(raw16.void_data(), 16);

    // TODO(ios): __attribute__((aligned(X))) with X >= 128 does not works on
    // the stack when building for arm64 on iOS, http://crbug.com/349003
#if !(defined(OS_IOS) && defined(ARCH_CPU_ARM64))
    EXPECT_ALIGNED(raw128.void_data(), 128);

    // NaCl x86-64 compiler emits non-validating instructions for >128
    // bytes alignment.
    // http://www.chromium.org/nativeclient/design-documents/nacl-sfi-model-on-x86-64-systems
    // TODO(hamaji): Ideally, NaCl compiler for x86-64 should workaround
    // this limitation and this #if should be removed.
    // https://code.google.com/p/nativeclient/issues/detail?id=3463
#if !(defined(OS_NACL) && defined(ARCH_CPU_X86_64))
    AlignedMemory<8, 256> raw256;
    EXPECT_EQ(256u, ALIGNOF(raw256));
    EXPECT_ALIGNED(raw256.void_data(), 256);

    // TODO(ios): This test hits an armv7 bug in clang. crbug.com/138066
#if !(defined(OS_IOS) && defined(ARCH_CPU_ARM_FAMILY))
    AlignedMemory<8, 4096> raw4096;
    EXPECT_EQ(4096u, ALIGNOF(raw4096));
    EXPECT_ALIGNED(raw4096.void_data(), 4096);
#endif // !(defined(OS_IOS) && defined(ARCH_CPU_ARM_FAMILY))
#endif // !(defined(OS_NACL) && defined(ARCH_CPU_X86_64))
#endif // !(defined(OS_IOS) && defined(ARCH_CPU_ARM64))
}

TEST(AlignedMemoryTest, DynamicAllocation)
{
    void* p = base::AlignedAlloc(8, 8);
    EXPECT_TRUE(p);
    EXPECT_ALIGNED(p, 8);
    base::AlignedFree(p);

    p = base::AlignedAlloc(8, 16);
    EXPECT_TRUE(p);
    EXPECT_ALIGNED(p, 16);
    base::AlignedFree(p);

    p = base::AlignedAlloc(8, 256);
    EXPECT_TRUE(p);
    EXPECT_ALIGNED(p, 256);
    base::AlignedFree(p);

    p = base::AlignedAlloc(8, 4096);
    EXPECT_TRUE(p);
    EXPECT_ALIGNED(p, 4096);
    base::AlignedFree(p);
}

TEST(AlignedMemoryTest, ScopedDynamicAllocation)
{
    std::unique_ptr<float, base::AlignedFreeDeleter> p(
        static_cast<float*>(base::AlignedAlloc(8, 8)));
    EXPECT_TRUE(p.get());
    EXPECT_ALIGNED(p.get(), 8);
}

} // namespace
